package com.haomin.searcher.indexer.util;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.FileFilter;
import java.util.LinkedList;
import java.util.List;

/**
 * @author haomin
 * @date 2022/10/04 18:07
 **/

@Slf4j    // 添加日志
@Service  // 注册成 Spring bean
public class FileScanner {
    public List<File> scanFile(String rootPath, FileFilter filter){
        List<File> resultList = new LinkedList<>();
        File rootFile = new File(rootPath);

        // 采用深度优先遍历，用递归完成
        traversal(rootFile, filter, resultList);
        return resultList;
    }

    private void traversal(File directoryFile, FileFilter filter, List<File> resultList) {
        // 1. 先通过，目录，找到该目录下文件有哪些
        File[] files = directoryFile.listFiles();
        if(files == null) {
            // 目录遍历完，或者该目录有问题（权限问题）
            return;
        }

        // 2. 遍历每个文件，判断是否符合要求
        for(File file : files){
            if(filter.accept(file)){
                resultList.add(file);
            }
        }

        // 3. 遍历每个文件，如果是目录，则进行深度优先遍历-递归
        for(File file : files){
            if(file.isDirectory()){
                traversal(file, filter, resultList);
            }
        }
    }
}